package com.github.niefy.modules.oss.cloud;
import com.github.niefy.common.exception.RRException;
import com.qiniu.http.Response;
import com.qiniu.storage.Configuration;
import com.qiniu.storage.Region;
import com.qiniu.storage.UploadManager;
import com.qiniu.util.Auth;
import org.apache.commons.io.IOUtils;

import java.io.IOException;
import java.io.InputStream;
/**
 * 七牛云存储
 *
 * @author Mark sunlightcs@gmail.com
 */
public class QiniuAbstractCloudStorageService extends AbstractCloudStorageService {
  private UploadManager uploadManager;
  private String token;
  public QiniuAbstractCloudStorageService(CloudStorageConfig config) {
    this.config = config;
    //初始化
    init();
  }
  private void init() {
    uploadManager = new UploadManager(new Configuration(Region.autoRegion()));
    token = Auth.create(config.getQiniuAccessKey(), config.getQiniuSecretKey()).uploadToken(config.getQiniuBucketName());
  }
  @Override
  public String upload(byte[] data, String path) {
    try {
      Response res = uploadManager.put(data, path, token);
      if (!res.isOK()) {
        throw new RuntimeException("上传七牛出错：" + res);
      }
    }
    catch (Exception e) {
      throw new RRException("上传文件失败，请核对七牛配置信息", e);
    }
    return config.getQiniuDomain() + "/" + path;
  }
  @Override
  public String upload(InputStream inputStream, String path) {
    try {
      byte[] data = IOUtils.toByteArray(inputStream);
      return this.upload(data, path);
    }
    catch (IOException e) {
      throw new RRException("上传文件失败", e);
    }
  }
  @Override
  public String uploadSuffix(byte[] data, String suffix) {
    return upload(data, getPath(config.getQiniuPrefix(), suffix));
  }
  @Override
  public String uploadSuffix(InputStream inputStream, String suffix) {
    return upload(inputStream, getPath(config.getQiniuPrefix(), suffix));
  }
}
